import cv2
import torch

img1=cv2.imread(r'C:\Users\admin\Desktop\train_set\image1\00001.png')
img2=cv2.imread(r'C:\Users\admin\Desktop\train_set\image2\00001.png')
img1=torch.tensor(img1)
img2=torch.tensor(img2)
img = torch.cat((img1,img2),2)
print()